home *** CD-ROM | disk | FTP | other *** search
/ Mac Easy 2010 May / Mac Life Ubuntu.iso / casper / filesystem.squashfs / usr / share / pyshared / DistUpgrade / DistUpgradeFetcherCore.py < prev    next >
Encoding:
Python Source  |  2009-04-27  |  10.6 KB  |  277 lines
  1. # DistUpgradeFetcherCore.py 
  2. #  
  3. #  Copyright (c) 2006 Canonical
  4. #  
  5. #  Author: Michael Vogt <michael.vogt@ubuntu.com>
  6. # 
  7. #  This program is free software; you can redistribute it and/or 
  8. #  modify it under the terms of the GNU General Public License as 
  9. #  published by the Free Software Foundation; either version 2 of the
  10. #  License, or (at your option) any later version.
  11. # 
  12. #  This program is distributed in the hope that it will be useful,
  13. #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15. #  GNU General Public License for more details.
  16. # 
  17. #  You should have received a copy of the GNU General Public License
  18. #  along with this program; if not, write to the Free Software
  19. #  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  20. #  USA
  21.  
  22. from string import Template
  23. import os
  24. import apt_pkg
  25. import apt
  26. import tarfile
  27. import socket
  28. import urlparse
  29. import urllib2
  30. import tempfile
  31. import shutil
  32. import sys
  33. import GnuPGInterface
  34. from gettext import gettext as _
  35. from aptsources.sourceslist import SourcesList
  36.  
  37. from utils import *
  38.  
  39. class DistUpgradeFetcherCore(object):
  40.     " base class (without GUI) for the upgrade fetcher "
  41.  
  42.     DEFAULT_MIRROR="http://archive.ubuntu.com/ubuntu"
  43.     DEFAULT_COMPONENT="main"
  44.     DEBUG = "DEBUG_UPDATE_MANAGER" in os.environ
  45.  
  46.     def __init__(self, new_dist, progress):
  47.         self.new_dist = new_dist
  48.         self.current_dist_name = get_dist()
  49.         self._progress = progress
  50.         # options to pass to the release upgrader when it is run
  51.         self.run_options = []
  52.  
  53.     def _debug(self, msg):
  54.         " helper to show debug information "
  55.         if self.DEBUG:
  56.             sys.stderr.write(msg+"\n")
  57.  
  58.     def showReleaseNotes(self):
  59.         return True
  60.  
  61.     def error(self, summary, message):
  62.         """ dummy implementation for error display, should be overwriten
  63.             by subclasses that want to more fancy method
  64.         """
  65.         print summary
  66.         print message
  67.         return False
  68.  
  69.     def authenticate(self):
  70.         if self.new_dist.upgradeToolSig:
  71.             f = self.tmpdir+"/"+os.path.basename(self.new_dist.upgradeTool)
  72.             sig = self.tmpdir+"/"+os.path.basename(self.new_dist.upgradeToolSig)
  73.             print "authenticate '%s' against '%s' " % (os.path.basename(f),os.path.basename(sig))
  74.             if not self.gpgauthenticate(f, sig):
  75.                 return False
  76.  
  77.         # we may return False here by default if we want to make a sig
  78.         # mandatory
  79.         return True
  80.  
  81.     def gpgauthenticate(self, file, signature,
  82.                         keyring='/etc/apt/trusted.gpg'):
  83.         """ authenticated a file against a given signature, if no keyring
  84.             is given use the apt default keyring
  85.         """
  86.         gpg = GnuPGInterface.GnuPG()
  87.         gpg.options.extra_args = ['--no-options',
  88.                                   '--homedir',self.tmpdir,
  89.                                   '--no-default-keyring',
  90.                                   '--ignore-time-conflict',
  91.                                   '--keyring', keyring]
  92.         proc = gpg.run(['--verify', signature, file],
  93.                        create_fhs=['status','logger','stderr'])
  94.         gpgres = proc.handles['status'].read()
  95.         try:
  96.             proc.wait()
  97.         except IOError,e:
  98.             # gnupg returned a problem (non-zero exit)
  99.             print "exception from gpg: %s" % e
  100.             print "Debug information: "
  101.             print proc.handles['status'].read()
  102.             print proc.handles['stderr'].read()
  103.             print proc.handles['logger'].read()
  104.             return False
  105.         if "VALIDSIG" in gpgres:
  106.             return True
  107.         print "invalid result from gpg:"
  108.         print gpgres
  109.         return False
  110.  
  111.     def extractDistUpgrader(self):
  112.           # extract the tarbal
  113.           fname = os.path.join(self.tmpdir,os.path.basename(self.uri))
  114.           print "extracting '%s'" % os.path.basename(fname)
  115.           if not os.path.exists(fname):
  116.               return False
  117.           try:
  118.               tar = tarfile.open(self.tmpdir+"/"+os.path.basename(self.uri),"r")
  119.               for tarinfo in tar:
  120.                   tar.extract(tarinfo)
  121.               tar.close()
  122.           except tarfile.ReadError, e:
  123.               logging.error("failed to open tarfile (%s)" % e)
  124.               return False
  125.           return True
  126.  
  127.     def verifyDistUprader(self):
  128.         # FIXME: check a internal dependency file to make sure
  129.         #        that the script will run correctly
  130.           
  131.         # see if we have a script file that we can run
  132.         self.script = script = "%s/%s" % (self.tmpdir, self.new_dist.name)
  133.         if not os.path.exists(script):
  134.             return self.error(_("Could not run the upgrade tool"),
  135.                          _("Could not run the upgrade tool") + ".  " + _("This is most likely a bug in the upgrade tool. "
  136.                           "Please report it as a bug"))
  137.         return True
  138.  
  139.     def mirror_from_sources_list(self, uri, default_uri):
  140.       """
  141.       try to figure what the mirror is from current sources.list
  142.  
  143.       do this by looing for matching DEFAULT_COMPONENT, current dist
  144.       in sources.list and then doing a http HEAD/ftp size request
  145.       to see if the uri is available on this server
  146.       """
  147.       self._debug("mirror_from_sources_list: %s" % self.current_dist_name)
  148.       sources = SourcesList(withMatcher=False)
  149.       seen = set()
  150.       for e in sources.list:
  151.         if e.disabled or e.invalid or not e.type == "deb":
  152.           continue
  153.         # check if we probed this mirror already
  154.         if e.uri in seen:
  155.           continue
  156.         # we are using the main mirror already, so we are fine
  157.         if (e.uri.startswith(default_uri) and 
  158.             e.dist == self.current_dist_name and
  159.             self.DEFAULT_COMPONENT in e.comps):
  160.           return uri
  161.         elif (e.dist == self.current_dist_name and
  162.               "main" in e.comps):
  163.           mirror_uri = e.uri+uri[len(default_uri):]
  164.           if url_downloadable(mirror_uri, self._debug):
  165.             return mirror_uri
  166.           seen.add(e.uri)
  167.       self._debug("no mirror found")
  168.       return ""
  169.  
  170.     def _expandUri(self, uri):
  171.         """
  172.         expand the uri so that it uses a mirror if the url starts
  173.         with a well know string (like archive.ubuntu.com)
  174.         """
  175.         # try to guess the mirror from the sources.list
  176.         if uri.startswith(self.DEFAULT_MIRROR):
  177.           self._debug("trying to find suitable mirror")
  178.           new_uri = self.mirror_from_sources_list(uri, self.DEFAULT_MIRROR)
  179.           if new_uri:
  180.             return new_uri
  181.         # if that fails, use old method
  182.         uri_template = Template(uri)
  183.         m = country_mirror()
  184.         new_uri = uri_template.safe_substitute(countrymirror=m)
  185.         # be paranoid and check if the given uri is really downloadable
  186.         try:
  187.             if not url_downloadable(new_uri, self._debug):
  188.               raise Exception("failed to download %s" % new_uri)
  189.         except Exception,e:
  190.             self._debug("url '%s' could not be downloaded" % e)
  191.             # else fallback to main server
  192.             new_uri = uri_template.safe_substitute(countrymirror='')
  193.         return new_uri
  194.  
  195.     def fetchDistUpgrader(self):
  196.         " download the tarball with the upgrade script "
  197.         self.tmpdir = tmpdir = tempfile.mkdtemp()
  198.         os.chdir(tmpdir)
  199.         # turn debugging on here (if required)
  200.         #apt_pkg.Config.Set("Debug::Acquire::http","1")
  201.         fetcher = apt_pkg.GetAcquire(self._progress)
  202.         if self.new_dist.upgradeToolSig != None:
  203.             uri = self._expandUri(self.new_dist.upgradeToolSig)
  204.             af = apt_pkg.GetPkgAcqFile(fetcher,uri, descr=_("Upgrade tool signature"))
  205.         if self.new_dist.upgradeTool != None:
  206.             self.uri = self._expandUri(self.new_dist.upgradeTool)
  207.             af = apt_pkg.GetPkgAcqFile(fetcher,self.uri, descr=_("Upgrade tool"))
  208.             if fetcher.Run() != fetcher.ResultContinue:
  209.                 return False
  210.             # check that both files are really there and non-null
  211.             for f in [os.path.basename(self.new_dist.upgradeToolSig),
  212.                       os.path.basename(self.new_dist.upgradeTool)]:
  213.                 if not (os.path.exists(f) and os.path.getsize(f) > 0):
  214.                     return False
  215.             return True
  216.         return False
  217.  
  218.     def runDistUpgrader(self):
  219.         #print "runing: %s" % script
  220.         args = [self.script]+self.run_options
  221.         if os.getuid() != 0:
  222.             os.execv("/usr/bin/sudo",["sudo"]+args)
  223.         else:
  224.             os.execv(self.script,args)
  225.  
  226.     def cleanup(self):
  227.       # cleanup
  228.       os.chdir("..")
  229.       # del tmpdir
  230.       shutil.rmtree(self.tmpdir)
  231.  
  232.     def run(self):
  233.         # see if we have release notes
  234.         if not self.showReleaseNotes():
  235.             return
  236.         if not self.fetchDistUpgrader():
  237.             self.error(_("Failed to fetch"),
  238.                   _("Fetching the upgrade failed. There may be a network "
  239.                     "problem. "))
  240.             return
  241.         if not self.extractDistUpgrader():
  242.             self.error(_("Failed to extract"),
  243.                   _("Extracting the upgrade failed. There may be a problem "
  244.                   "with the network or with the server. "))
  245.                   
  246.             return
  247.         if not self.verifyDistUprader():
  248.             self.error(_("Verfication failed"),
  249.                   _("Verifying the upgrade failed.  There may be a problem "
  250.                     "with the network or with the server. "))
  251.             self.cleanup()
  252.             return
  253.         if not self.authenticate():
  254.             self.error(_("Authentication failed"),
  255.                   _("Authenticating the upgrade failed. There may be a problem "
  256.                     "with the network or with the server. "))
  257.             self.cleanup()
  258.             return
  259.         try:
  260.           self.runDistUpgrader()
  261.         except OSError, e:
  262.           if e.errno == 13:
  263.             self.error(_("Can not run the upgrade"),
  264.                        _("This usually is caused by a system were /tmp "
  265.                          "is mounted noexec. Please remount without "
  266.                          "noexec and run the upgrade again."))
  267.             return False
  268.           else:
  269.             self.error(_("Can not run the upgrade"),
  270.                        _("The error message is '%s'." % e.strerror))
  271.         return True
  272.  
  273. if __name__ == "__main__":
  274.     d = DistUpgradeFetcherCore(None,None)
  275. #    print d.authenticate('/tmp/Release','/tmp/Release.gpg')
  276.     print "got mirror: '%s'" % d.mirror_from_sources_list("http://archive.ubuntu.com/ubuntu/dists/intrepid-proposed/main/dist-upgrader-all/0.93.34/intrepid.tar.gz", "http://archive.ubuntu.com/ubuntu")
  277.